/*
Continuous Testing Toolkit
Copyright (C) 2009 Paul R. Holser, Jr.

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*/

package org.infinitest.toolkit.filter;

import java.util.Set;

import static com.google.common.collect.Sets.*;
import org.junit.runner.Description;
import org.junit.runner.notification.RunNotifier;
import org.junit.runner.notification.StoppedByUserException;

/**
 * Notifier that keeps track of what test methods are executed.
 */
class CollectingRunNotifier extends RunNotifier {
    Set<String> executedTestMethods = newLinkedHashSet();

    @Override
    public void fireTestStarted(Description description) throws StoppedByUserException {
        String displayName = description.getDisplayName();
        executedTestMethods.add(extractMethodName(displayName));
    }

    private String extractMethodName(String displayName) {
        return displayName.substring(0, displayName.indexOf("("));
    }

    boolean containsTest(String testName) {
        return executedTestMethods.contains(testName);
    }

    boolean isEmpty() {
        return executedTestMethods.isEmpty();
    }
}
